from mpi4py import MPI
comm = MPI.COMM_WORLD
rank = comm.Get_rank()
a = 3.0
b = 2.0
if rank == 0:
    print(f'Process {rank}: {a + b}')
elif rank == 1:
    print(f'Process {rank}: {a * b}')
elif rank == 2:
    print(f'Process {rank}: {max(a, b)}')
